# 路径 被定义为一条从树中任意节点出发，沿父节点-子节点连接，达到任意节点的序列。同一个节点在一条路径序列中 至多出现一次 。
# 该路径 至少包含一个 节点，且不一定经过根节点。
#  路径和 是路径中各节点值的总和。
#
#  给你一个二叉树的根节点 root ，返回其 最大路径和 。
#
#  示例 1：
# 输入：root = [1,2,3]
# 输出：6
# 解释：最优路径是 2 -> 1 -> 3 ，路径和为 2 + 1 + 3 = 6
#
#  示例 2：
# 输入：root = [-10,9,20,null,null,15,7]
# 输出：42
# 解释：最优路径是 15 -> 20 -> 7 ，路径和为 15 + 20 + 7 = 42
import sys
from com.example.tree.tree_node import TreeNode
from typing import Optional


class Solution:
    def maxPathSum(self, root: Optional[TreeNode]) -> int:
        self.res = -sys.maxsize - 1
        def dfs(root: Optional[TreeNode]) -> int:
            """
            求每个节点的最大贡献值
            :param root:
            :return:
            """
            if not root:
                return 0
            leftMaxGain = max(dfs(root.left), 0)  # 左子节点只有在最大贡献值大于 0 时，才可能会被选取
            rightMaxGain = max(dfs(root.right), 0)  # 右子节点只有在最大贡献值大于 0 时，才可能会被选
            self.res = max(self.res, root.val + leftMaxGain + rightMaxGain)  # 尝试更新结果
            return root.val + max(leftMaxGain, rightMaxGain)  # 当前节点的最大贡献值只能取左右子树的较大的贡献值与当前节点的val之和
        dfs(root)
        return self.res


if __name__ == "__main__":
    #     3
    #    / \
    #   9  20
    #     /  \
    #    15   7
    root = TreeNode(3)
    root.left, root.right = TreeNode(9), TreeNode(20)
    root.right.left, root.right.right = TreeNode(15), TreeNode(7)
    print(Solution().maxPathSum(root))
    #    -10
    #    /  \
    #   9   20
    #      /  \
    #     15   7
    root = TreeNode(-10)
    root.left, root.right = TreeNode(9), TreeNode(20)
    root.right.left, root.right.right = TreeNode(15), TreeNode(7)
    print(Solution().maxPathSum(root))
